import os
def dir_size(dir):
    '''
    定义计算指定目录大小的函数
    '''
    size = 0
    #判断指定目录是否为文件
    if os.path.isfile(dir):
        size+=os.path.getsize(dir)
    #判断指定目录是否为文件夹
    if os.path.isdir(dir):
        dir_list = os.listdir(dir)
        for f in dir_list:
            file = os.path.join(dir,f)
            if os.path.isfile(file):
                size+=os.path.getsize(file)
            if os.path.isdir(file):
                size+=dir_size(file)#递归统计
    return size
print(dir_size('./ce'))


